/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package br.com.pedidos.servidor.mb.model;

import br.com.pedidos.servidor.entity.Pedido;
import java.io.Serializable;
import java.util.List;
import javax.faces.model.ListDataModel;
import org.primefaces.model.SelectableDataModel;

/**
 *
 * @author Weslley
 */
public class PedidoDataModel extends ListDataModel<Pedido> implements SelectableDataModel<Pedido>, Serializable {

    public PedidoDataModel() {

    }

    public PedidoDataModel(List<Pedido> data) {
        super(data);
    }

    @Override
    public Object getRowKey(Pedido p) {
        return p.getId();
    }

    @Override
    public Pedido getRowData(String rowKey) {
        List<Pedido> pedidos = (List<Pedido>) getWrappedData();

        for (Pedido p : pedidos) {
            if (p.getId() == Long.parseLong(rowKey)) {
                return p;
            }
        }

        return null;
    }
}
